horror_df

CI <- function(data, coverage_prob){ 
  #Generates a normal prediction interval with an intended coverage probability of coverage_prob based on a vector of numeric data
  lower_zscore <- qnorm((1-coverage_prob)/2)
  upper_zscore <- qnorm(((1-coverage_prob)/2) + coverage_prob)
  avg <- mean(data)
  stan_d <- sd(data)
  lower_bound <- avg + lower_zscore*stan_d
  upper_bound <- avg + upper_zscore*stan_d
  return(data.frame(PI_percentage = coverage_prob, lower = lower_bound, upper = upper_bound))
}

one_beta_simulation <- function(n, alpha, beta, ci_prop){
  #Assesses prediction accuracy and actual coverage probability of a normal prediction interval when used on a vector of numeric data of size n. The numeric data is generated from a beta distribution with parameters alpha and beta.
  
  cover_df <- CI(rbeta(n, alpha, beta), ci_prop)
  cover_prop <- pbeta(cover_df[1, "upper"], alpha, beta) - pbeta(cover_df[1, "lower"], alpha, beta)
  mean_in_interval <- .5 >= cover_df[1, "lower"] & .5 <= cover_df[1,"upper"]
  param_df <- data.frame(cover = cover_prop, alpha = rep(alpha, nrow(cover_df)), beta = rep(beta, nrow(cover_df)), mean_in_interval = mean_in_interval)
  df <- cbind(cover_df, param_df)
  return(df)
}

beta_sims_n <- function(n){
  #Iterates over a vector of possible alpha = beta values and applies one_beta_simulation to each possible value of alpha/beta. All simulations use data of sample size n.
  df1 <- map(parameters,\(param) one_beta_simulation(n, param, param, ci) ) %>%
  list_rbind()
  df2 <- data.frame(n = rep(n, nrow(df1)))
  df <- cbind(df2, df1)
  return(df)
}